from .SAPNet import SAPNet
from .DA_Net import DA_Net
from .DA_ICR_CCR import DA_ICR_CCR
from .SW_Net import SW_Net

DETECTORS = {
    'SAPNet': SAPNet,
    'DA_Net': DA_Net,
    'DA_ICR_CCR': DA_ICR_CCR,
    'SW_Net': SW_Net
}


def build_detectors(cfg):
    model_name = cfg.MODEL.MODEL_NAME
    model = DETECTORS[model_name](cfg)
    return model
